package leetcode.pre50;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，
 * <p>
 * 并返回他们的数组下标。
 * <p>
 * 你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。
 * <p>
 * 示例:
 * <p>
 * 给定 nums = [2, 7, 11, 15], target = 9
 * <p>
 * 因为 nums[0] + nums[1] = 2 + 7 = 9
 * 所以返回 [0, 1]
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/two-sum
 * 其余见{@link leetcode.a_总结分布.Code11_N数之和}
 * @date 2020/3/17 22:20
 */
public class Code01_TwoNumSum {
    //方式1：两次遍历
    //TODO BUG 已经打乱了顺序 方式2：排序，然后利用两个指针p1,p2一个指向头一个指向尾巴。
    //方式3：利用HashMap，key是值，V是记录的索引。

    /**
     * 排序不行，已经打乱了顺序。如果返回boolean可以用排序
     */
    public int[] twoSum1(int[] nums, int target) {
        if (nums == null || nums.length <= 1) return null;
        Arrays.sort(nums);
        int p1 = 0;
        int p2 = nums.length - 1;
        while (p1 < p2) {
            if (nums[p1] + nums[p2] == target) {
                //TODO  此时的p1,p2不一定是原来的顺序，
                return new int[]{p1, p2};
            }
            if (nums[p1] + nums[p2] < target) {
                p1++;
            } else {
                p2--;
            }
        }
        return null;
    }

    public static int[] twoSum2(int[] nums, int target) {
        if (nums == null || nums.length <= 1) return null;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            map.put(nums[i], i);
        }
        for (int i = 0; i < nums.length; i++) {
            int temp = target - nums[i];
            //    if(map.containsKey(temp)){  BUG  {3,2,4} target = 6 输出{0,0}，应该不包含自身
            //    if(map.containsKey(temp) && temp != i){ BUG  {3,3} target = 6 输出null ,map此时(k,v)=(3,1)
            //这样写，如果数组有重复数字，那么后面的key会把前面的覆盖掉。此时map.get(temp) 与i不等。
            if (map.containsKey(temp) && map.get(temp) != i) {
                return new int[]{i, map.get(temp)};
            }
        }
        return null;
    }

    //方式4：一次遍历hash
    public static int[] twoSum3(int[] nums, int target) {
        if (nums == null || nums.length <= 1) return null;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int temp = target - nums[i];
            if (map.containsKey(temp)) {//map含有的肯定是之前的，不包含当前i位置，因为判断完才放入
                return new int[]{map.get(temp), i};
            }
            map.put(nums[i], i);
        }
        return null;
    }

    public static void main(String[] args) {
        int[] ints = twoSum2(new int[]{1, 1, 3, 3}, 4);
        System.out.println(Arrays.toString(ints));
    }
}
